/*
 * Copyright (C) 2017 Twitter, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

#import "TWTRTestImageLoaderCache.h"

@interface TWTRTestImageLoaderCache ()

@property (nonatomic, readonly) NSMutableDictionary *imageFixtures;

@end

@implementation TWTRTestImageLoaderCache

- (instancetype)initWithImageFixturesDictionary:(NSDictionary *)imageFixturesDictionary
{
    if (self = [super init]) {
        _imageFixtures = [imageFixturesDictionary mutableCopy];
    }

    return self;
}

- (void)setImage:(UIImage *)image forKey:(NSString *)key
{
    self.imageFixtures[key] = image;
}

- (void)setImageData:(NSData *)imageData forKey:(NSString *)key
{
    UIImage *image = [UIImage imageWithData:imageData];
    [self setImage:image forKey:key];
}

- (UIImage *)fetchImageForKey:(NSString *)key
{
    return self.imageFixtures[key];
}

- (UIImage *)removeImageForKey:(NSString *)key
{
    UIImage *image = [self fetchImageForKey:key];
    [self.imageFixtures removeObjectForKey:key];
    return image;
}

- (void)removeAllImages
{
    [self.imageFixtures removeAllObjects];
}

@end
